from .RainTrainH import get_raintrainh_loader

def build_loader(args):
    if args.datasets == "raintrainh":
        label_loader, unlabel_loader, test_loader = get_raintrainh_loader(
            root=args.data_path,
            train_crop_size=args.train_crop_size,
            batch_size=args.batch_size,
        )
        return train_loader, test_loader
  
    else:
        raise NotImplementedError



